在二分类问题中,输出$y\in \{0, 1\}$。同样的,我们也可以用线性拟合来尝试解决二分类问题(如下图左),但数据点比较异常时,容易出现下图右这样的情况:
一般,在二分类问题中,我们会选用『logistic函数』来拟合(因为形状像S,又称为『sigmoid函数』):
logistic函数$g(z)=1/(1+e^{-z})$的形状如下:
可以定义
于是:
进行极大似然估计:
为了计算方便,定义
利用梯度上升进行求解:
其中
最终的梯度上升结果几乎与线性拟合中的梯度下降结果一样。